import java.util.TreeSet;

/**
 * 给你一个整数数组 nums 和两个整数 k 和 t 。请你判断是否存在 两个不同下标 i 和 j，使得 abs(nums[i] - nums[j]) <= t ，同时又满足 abs(i - j) <= k 。
 * <p>
 * 如果存在则返回 true，不存在返回 false。
 * <p>
 *  
 * <p>
 * 示例 1：
 * <p>
 * 输入：nums = [1,2,3,1], k = 3, t = 0
 * 输出：true
 * 示例 2：
 * <p>
 * 输入：nums = [1,0,1,1], k = 1, t = 2
 * 输出：true
 * 示例 3：
 * <p>
 * 输入：nums = [1,5,9,1,5,9], k = 2, t = 3
 * 输出：false
 *  
 * <p>
 * 提示：
 * <p>
 * 0 <= nums.length <= 2 * 104
 * -231 <= nums[i] <= 231 - 1
 * 0 <= k <= 104
 * 0 <= t <= 231 - 1
 */
class Solution {

    public static void main(String[] args) {
        System.out.println(containsNearbyAlmostDuplicate(new int[]{1, 3, 5, 7, 9, 100, 101, 15, 17, 19}, 3, 1));
        System.out.println(containsNearbyAlmostDuplicate(new int[]{3, 6, 0, 2}, 2, 2));
        System.out.println(containsNearbyAlmostDuplicate(new int[]{-2147483648, 2147483647}, 1, 1));
        System.out.println(containsNearbyAlmostDuplicate(new int[]{1, 5, 9, 1, 5, 9}, 2, 3));
        System.out.println(containsNearbyAlmostDuplicate(new int[]{1, 0, 1, 1}, 1, 2));
        System.out.println(containsNearbyAlmostDuplicate(new int[]{1, 2, 3, 1}, 3, 0));
    }

    /**
     * 滑动窗口+有序集合，可以快速定位新的元素是不是满足条件的
     * floor(E e) 方法返回在这个集合中小于或者等于给定元素的最大元素，如果不存在这样的元素,返回null.
     * ceiling(E e) 方法返回在这个集合中大于或者等于给定元素的最小元素，如果不存在这样的元素,返回null.
     *
     * @param nums
     * @param k
     * @param t
     * @return
     */
    public static boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        int n = nums.length;
        TreeSet<Long> set = new TreeSet<Long>();
        for (int i = 0; i < n; i++) {
            Long ceiling = set.ceiling((long) nums[i] - (long) t);
            if (ceiling != null && ceiling <= (long) nums[i] + (long) t) {
                return true;
            }
            set.add((long) nums[i]);
            if (i >= k) {
                // 如果是最后了，移除之前加入的元素
                set.remove((long) nums[i - k]);
            }
        }
        return false;
    }


    /**
     * 滑动窗口的思想
     *
     * @param nums
     * @param k
     * @param t
     * @return
     */
    public static boolean containsNearbyAlmostDuplicate1(int[] nums, int k, int t) {
        if (k == 0 || t < 0 || nums.length < 1) {
            return false;
        }
        if (t == 0) {
            // 当t==0的时候特殊处理
            return containsNearbyAlmostDuplicate(nums, k);
        }
        k = Math.min(k, nums.length - 1);
        // 处理前k个，放到有序队列
        for (int i = 0; i < k; i++) {
            for (int j = i + 1; j <= k; j++) {
                if (Math.abs((long) nums[i] - (long) nums[j]) <= t) {
                    return true;
                }
            }
        }
        for (int i = 1; i <= nums.length - k - 1; i++) {
            for (int j = 0; j < k; j++) {
                // 后面k个判断
                if (Math.abs((long) nums[i + j] - (long) nums[i + k]) <= t) {
                    return true;
                }
            }
        }
        return false;
    }

    public static boolean containsNearbyAlmostDuplicate(int[] nums, int k) {
        k = Math.min(k, nums.length - 1);
        for (int i = 0; i < k; i++) {
            for (int j = i + 1; j <= k; j++) {
                if (nums[i] == nums[j]) {
                    return true;
                }
            }
        }
        for (int i = 1; i <= nums.length - k - 1; i++) {
            if (nums[i] == nums[i + k]) {
                return true;
            }
        }
        for (int i = nums.length - k; i < nums.length; i++) {
            for (int j = i + 1; j < nums.length; j++) {
                if (nums[i] == nums[j]) {
                    return true;
                }
            }
        }
        return false;
    }

}
